package com.tgy.哈希表;

import java.util.Scanner;

/***
 * @ClassName: HashTabDemo
 * @Description: 哈希散列测试
 * @Auther: tanggy3
 * @Date: 16:06 2022/11/22
 * @version : V1.0
 */

public class HashTabDemo {

    public static void main(String[] args) {

        //创建哈希表
        HashTab hashTab = new HashTab(7);

        //写一个简单的菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while (true) {
            System.out.println("add:  添加雇员");
            System.out.println("list: 显示雇员");
            System.out.println("find: 查找雇员");
            System.out.println("exit: 退出系统");

            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入名字");
                    String name = scanner.next();
                    //创建 雇员
                    Emp emp = new Emp(id, name);
                    hashTab.add(emp);
                    break;
                case "list":
                    hashTab.list();
                    break;
                case "find":
                    System.out.println("请输入要查找的id");
                    id = scanner.nextInt();
                    hashTab.findEmpById(id);
                    break;
                case "del":
                    System.out.println("请输入要删除的id");
                    id = scanner.nextInt();
                    hashTab.del(id);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);

                default:
                    break;
            }
        }

    }

}

//创建HashTab 管理多条链表，通过散列函数决定数据具体到哪一个链表
class HashTab {
    private int size;//有几个链表 决定散列函数的分配
    private EmpLinkedList[] empLinkedListArray;//链表数组 存放链表

    //构造函数
    public HashTab(int size) {
        this.size = size;
        //初始化存放链表的数组
        empLinkedListArray = new EmpLinkedList[size];
        for (int i = 0; i < size; i++) {
            empLinkedListArray[i] = new EmpLinkedList();
        }
    }

    //散列函数
    private int hashFun(int no){
        return no%size;
    }

    //添加雇员
    public void add(Emp emp) {
        //根据员工的id ,得到该员工应当添加到哪条链表
        int empLinkedListNO = hashFun(emp.id);
        //将emp 添加到对应的链表中
        empLinkedListArray[empLinkedListNO].add(emp);

    }
    //遍历所有的链表,遍历hashtab
    public void list() {
        for(int i = 0; i < size; i++) {
            empLinkedListArray[i].list(i);
        }
    }

    public void findEmpById(int id){
        int i = hashFun(id);
        Emp emp = empLinkedListArray[i].find(id);
        if(emp != null) {//找到
            System.out.printf("在第%d条链表中找到 雇员 id = %d\n", (i), id);
        }else{
            System.out.println("在哈希表中，没有找到该雇员~");
        }
    }

    public void del(int id){
        int i = hashFun(id);
        Emp emp = empLinkedListArray[i].del(id);
        if(emp != null) {//找到
            System.out.printf("成功删除，在第%d条链表中找到 雇员 id = %d\n", (i), id);
        }else{
            System.out.println("在哈希表中，没有找到该雇员~");
        }
    }
}

//链表，链表里面就是各个雇员信息
class EmpLinkedList {
    private Emp head;

    public void add(Emp emp) {
        Emp curEmp = head;
        if(head == null) {
            head = emp;
            return;
        }
        //找到末尾节点,这里要判断末尾的前一个，不然出不来
        while (curEmp.next!=null ){
            curEmp = curEmp.next;
        }
        //添加雇员
        curEmp.next = emp;
        System.out.println("添加完成");

    }

    //传入链表编号，方便输出文字格式化
    public void list(int empLinkedListNo) {
        if(head == null) { //说明链表为空
            System.out.println("第 "+(empLinkedListNo)+" 条链表为空");
            return;
        }
        System.out.print("第 "+(empLinkedListNo)+" 条链表的信息为");
        Emp curEmp = head;
        while (curEmp!=null){
            System.out.printf(" => id=%d name=%s\t", curEmp.id, curEmp.name);
            curEmp = curEmp.next;
        }
        System.out.println();
    }


    //查找
    public Emp find(int id) {
        Emp reEmp = null;
        //链表为空不存在
        if (head == null){
            return reEmp;
        }
        Emp curEmp = head;
        while (curEmp.next!=null){
            if (curEmp.id == id){
                reEmp = curEmp;
                break;
            }else{
                curEmp = curEmp.next;
            }
        }
        return curEmp;

    }

    public Emp del(int no) {
        Emp reEmp = null;
        //链表为空不存在
        if (head == null){
            return reEmp;
        }
        Emp curEmp = head;
        while (curEmp.next!=null){
            if (curEmp.next.id == no){
                //把目标的下一个节点接入目标的上一个节点
                reEmp = curEmp.next;
                curEmp.next = curEmp.next.next;
                break;
            }else{
                curEmp = curEmp.next;
            }
        }
        return curEmp;
    }

}

//雇员
class Emp {
    public int id;
    public String name;
    public Emp next; //next 默认为 null

    public Emp(int id, String name) {
        this.id = id;
        this.name = name;
    }
}




